#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using System;
using System.Collections.Generic;
using System.Reflection;


namespace NUnitGrid.GridClient.Plugins.GridClient.NUnitTestsFinder
{
    public class FixtureFinder : IFixtureFinder
    {
        private readonly Assembly assembly;
        private readonly Type testFixtureAttribute;

        public FixtureFinder(Assembly assembly, Type testFixtureAttribute)
        {
            this.assembly = assembly;
            this.testFixtureAttribute = testFixtureAttribute;
        }

        public string[] GetTestFixtures()
        {
            var testFixtureNames = new List<string>();
            var types = assembly.GetTypes();
            foreach (var type in types)
            {
                var testFixtureAttributes = type.GetCustomAttributes(testFixtureAttribute, false);
                if (testFixtureAttributes.Length == 1)
                {
                    testFixtureNames.Add(type.FullName);
                }
            }
            return testFixtureNames.ToArray();
        }
    }
}